package com.example.bootkafka1.common;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.stereotype.Component;

import java.text.MessageFormat;

/**
 * Kafka 生产者
 */
@Component
public class KafkaProducer {

    private Logger logger = LoggerFactory.getLogger(KafkaProducer.class);

    @Autowired
    private KafkaTemplate<Object,Object> kafkaTemplate;

    private Gson gson = new GsonBuilder().create();

    /**
     * 推送数据
     * @param topic  主题名称
     * @param object 数据
     *  当get方法抛出一个错误时，说明数据没有被正确写入，此时需要处理这个错误。
     */
    public void send(String topic,Object object){
        try {
            kafkaTemplate.send(topic, gson.toJson(object)).get();
        } catch (Exception e) {
            logger.info(MessageFormat.format("推送数据错误：{0}",e.getMessage()));
            e.printStackTrace();
        }
    }


}
